package leetcode.leetcode23;

/**
 * @author czt
 * @version 1.0
 * @since 2024/12/18
 *
 * 合并多个有序链表
 */
public class Leetcode23 {

    public ListNode mergeTwoLists(ListNode list1, ListNode list2) {
        // 哨兵节点
        ListNode s = new ListNode(-1, null);

        // 新链表的头部
        ListNode p = s;

        while (list1 != null && list2 != null) {
            // 比较两个链表，那个小将它链到p后面；与此同时，p向后移，小的链表也向后移
            if (list1.val < list2.val) {
                p.next = list1;
                list1 = list1.next;
            } else {
                p.next = list2;
                list2 = list2.next;
            }
            p = p.next;
        }

        if (list1 != null) {
            p.next = list1;
        }
        if (list2 != null) {
            p.next = list2;
        }

        return s.next;
    }

    // 采用递归思想，将多个链表拆分为两个链表，调用上面的方法进行合并
    public ListNode mergeKLists(ListNode[] lists) {
        if (lists == null || lists.length == 0) {
            return null;
        }

        return split(lists, 0, lists.length - 1);
    }

    // 返回合并后的链表，i和j代表左右边界
    private ListNode split(ListNode[] lists, int i, int j) {
        if (i == j) { // 数组内只有一个链表时
            return lists[i];
        }
        // 取中间，相当于加起来除2
        int m = (i + j) >>> 1;
        ListNode left = split(lists, i, m);
        ListNode right = split(lists, m + 1, j);
        return mergeTwoLists(left, right);
    }
}
